from channels.generic.websocket import WebsocketConsumer
from channels.exceptions import StopConsumer

from utils.spark.gpt import test
class GptConsumer(WebsocketConsumer):
    def websocket_connect(self, message):
        # 有客户端来向后端发送websocket连接的请求时，自动触发。
        # 服务端允许和客户端创建连接。
        self.accept()

    def websocket_receive(self, message):
        # 浏览器基于websocket向后端发送数据，自动触发接收消息。
        print(message)
        self.send("##########start##########")
        # self.send("小天：")
        test.gpt(self, message['text'])
        self.send("##########end##########")
        # self.send("小天：\n\t" + gpt.main(message['text'])[-1]['content'])
        # Document_Q_And_A_mul.sendQuestion(self, message['text'])
        # self.close()

    def websocket_disconnect(self, message):
        # 客户端与服务端断开连接时，自动触发。
        print("断开连接")
        raise StopConsumer()
